home *** CD-ROM | disk | FTP | other *** search
/ Reverse Code Engineering RCE CD +sandman 2000 / ReverseCodeEngineeringRceCdsandman2000.iso / RCE / Tools / Win95 Secrets / SETUP.Z / W32SVFLT.C < prev    next >
Encoding:
C/C++ Source or Header  |  1995-07-19  |  5.9 KB  |  211 lines
  1. //==================================
  2. // W32SVSPY - Matt Pietrek 1995
  3. // FILE: W32SVFLT.C
  4. //==================================
  5. #include <windows.h>
  6. #include <stdio.h>
  7. #include <string.h>
  8. #pragma hdrstop
  9. #include "w32srvdb.h"
  10. #include "w32svspy.h"
  11. #include "w32svflt.h"
  12.  
  13. void UpdateFilterDialogServiceList( HWND hWndDlg, unsigned index );
  14. void HandleWin32ServiceListboxDblClk( HWND hWndDlg );
  15.  
  16. void Handle_FilterDlg_WM_COMMAND(HWND hWndDlg, WPARAM wParam, LPARAM lParam)
  17. {
  18.     switch ( LOWORD(wParam) )
  19.     {
  20.         case IDOK:
  21.  
  22.             // Fall through to IDCANCEL
  23.         case IDCANCEL:
  24.             EndDialog(hWndDlg, 0); return;
  25.             
  26.         case IDC_FILTER_LB_PROVIDERS:
  27.             if ( HIWORD(wParam) == LBN_SELCHANGE )
  28.             {
  29.                 DWORD currSel = SendMessage((HWND)lParam, LB_GETCURSEL, 0, 0);
  30.                 if ( currSel != LB_ERR )
  31.                     UpdateFilterDialogServiceList( hWndDlg, currSel );
  32.             }
  33.             break;
  34.             
  35.         case IDC_FILTER_LB_SERVICES:
  36.             if ( HIWORD(wParam) == LBN_DBLCLK )
  37.                 HandleWin32ServiceListboxDblClk( hWndDlg );
  38.             break;
  39.     }
  40. }
  41.  
  42. void Handle_FilterDlg_WM_INITDIALOG(HWND hWndDlg, WPARAM wParam, LPARAM lParam)
  43. {
  44.     unsigned i;
  45.     HWND hWndLb;
  46.     
  47.     hWndLb = GetDlgItem(hWndDlg,IDC_FILTER_LB_PROVIDERS);
  48.     
  49.     for ( i=0; i < CWin32ServiceVxDs; i++ )
  50.     {
  51.         SendMessage( hWndLb,
  52.                     LB_ADDSTRING,
  53.                     (WPARAM)0,
  54.                     (LPARAM)W32ServiceTable[i].pszVxDName );
  55.     }
  56.  
  57.     // Force the first VXD's services to be shown in the service list
  58.     SendMessage( hWndLb, LB_SETCURSEL, 0, 0 );
  59.     PostMessage( hWndDlg,
  60.                 WM_COMMAND,
  61.                 MAKEWPARAM(IDC_FILTER_LB_PROVIDERS, LBN_SELCHANGE),
  62.                 (LPARAM)hWndLb );
  63. }
  64.  
  65. void UpdateFilterDialogServiceList( HWND hWndDlg, unsigned index )
  66. {
  67.     unsigned i;
  68.     HWND hWndLb;
  69.     
  70.     hWndLb = GetDlgItem( hWndDlg, IDC_FILTER_LB_SERVICES );
  71.     
  72.     SendMessage( hWndLb, LB_RESETCONTENT, 0, 0 );
  73.     SendMessage( hWndLb, WM_SETREDRAW, FALSE, 0 );
  74.  
  75.     for ( i=0; i < W32ServiceTable[index].cServiceCalls; i++ )
  76.     {
  77.         char szBuffer[512];
  78.         
  79.         wsprintf(szBuffer, "%c %08X %s",
  80.                     W32ServiceTable[index].pServiceCalls[i].fIgnore ? '-': '+',
  81.                     W32ServiceTable[index].vxdID + i,
  82.                     W32ServiceTable[index].pServiceCalls[i].pszServiceName);
  83.  
  84.         SendMessage(hWndLb, LB_ADDSTRING, 0, (LPARAM)szBuffer );
  85.     }
  86.         
  87.     SendMessage( hWndLb, WM_SETREDRAW, TRUE, 0 );
  88. }
  89.  
  90. void HandleWin32ServiceListboxDblClk( HWND hWndDlg )
  91. {
  92.     DWORD currVxD;
  93.     DWORD currServ;
  94.     char szBuffer[512];
  95.     
  96.     currVxD = SendDlgItemMessage(hWndDlg, IDC_FILTER_LB_PROVIDERS,
  97.                                     LB_GETCURSEL, 0, 0);
  98.                             
  99.     if ( currVxD == LB_ERR )
  100.         return;
  101.  
  102.     currServ = SendDlgItemMessage( hWndDlg, IDC_FILTER_LB_SERVICES,
  103.                                     LB_GETCURSEL, 0, 0 );
  104.                         
  105.     if ( currServ == LB_ERR )
  106.         return;
  107.  
  108.     // Toggle the state of the fIgnore flag
  109.     W32ServiceTable[currVxD].pServiceCalls[currServ].fIgnore ^= 1;
  110.  
  111.     SendDlgItemMessage( hWndDlg, IDC_FILTER_LB_SERVICES, LB_GETTEXT, 
  112.                         currServ, (LPARAM)szBuffer );
  113.                     
  114.     szBuffer[0] = (szBuffer[0] == '+') ? '-' : '+';
  115.  
  116.     SendDlgItemMessage( hWndDlg, IDC_FILTER_LB_SERVICES, LB_DELETESTRING, 
  117.                         (WPARAM)currServ, 0 );
  118.     SendDlgItemMessage( hWndDlg, IDC_FILTER_LB_SERVICES, LB_INSERTSTRING, 
  119.                         (WPARAM)currServ, (LPARAM)szBuffer );
  120.     
  121. } 
  122.  
  123. BOOL CALLBACK W32SpyFilterDlgProc(HWND hWndDlg, UINT msg,
  124.                                     WPARAM wParam, LPARAM lParam)
  125. {
  126.     switch ( msg )
  127.     {
  128.         case WM_COMMAND:
  129.             Handle_FilterDlg_WM_COMMAND( hWndDlg, wParam, lParam );
  130.             break;
  131.  
  132.         case WM_INITDIALOG:
  133.             Handle_FilterDlg_WM_INITDIALOG( hWndDlg, wParam, lParam );
  134.             break;
  135.  
  136.         case WM_CLOSE:
  137.             EndDialog(hWndDlg, 0); return FALSE;
  138.     }
  139.     return FALSE;
  140. }
  141.  
  142. static void GetSavedFiltersFilename( PSTR pszBuffer )
  143. {
  144.     PSTR p;
  145.     
  146.     GetModuleFileName( 0, pszBuffer, MAX_PATH );
  147.     
  148.     p = strrchr( pszBuffer, '\\' );
  149.     strcpy( p+1, "w32svspy.flt" );
  150. }
  151.  
  152. BOOL SaveFilterValues(void)
  153. {
  154.     FILE * pFile;
  155.     char szSaveFileName[MAX_PATH];
  156.     unsigned i, j;
  157.     
  158.     GetSavedFiltersFilename( szSaveFileName );
  159.     pFile = fopen( szSaveFileName, "wb" );
  160.     if ( !pFile )
  161.     {
  162.         MessageBox(0, "Unable to open saved filters file!", 0, MB_OK);
  163.         return FALSE;
  164.     }
  165.     
  166.     for ( i=0; i < CWin32ServiceVxDs; i++ )
  167.     {
  168.         for ( j=0; j < W32ServiceTable[i].cServiceCalls; j++ )
  169.         {
  170.             if ( W32ServiceTable[i].pServiceCalls[j].fIgnore )
  171.             {
  172.                 DWORD serviceID;
  173.                 
  174.                 serviceID = W32ServiceTable[i].vxdID + j;
  175.                 fwrite( &serviceID, sizeof(serviceID), 1, pFile );
  176.             }
  177.         }
  178.     }
  179.     
  180.     fclose( pFile );
  181.  
  182.     return TRUE;
  183. }
  184.  
  185. BOOL LoadSavedFilterValues(void)
  186. {
  187.     FILE * pFile;
  188.     char szSaveFileName[MAX_PATH];
  189.     DWORD serviceID;
  190.     PWIN32_SERVICE_CALL pWin32ServiceCall;
  191.         
  192.     GetSavedFiltersFilename( szSaveFileName );
  193.     pFile = fopen( szSaveFileName, "rb" );
  194.     if ( !pFile )
  195.     {
  196.         // MessageBox(0, "Unable to open saved filters file!", 0, MB_OK);
  197.         return FALSE;
  198.     }
  199.     
  200.     while ( !feof(pFile) && fread(&serviceID, sizeof(serviceID), 1, pFile) )
  201.     {
  202.         pWin32ServiceCall = LookupWin32ServiceCall( serviceID );
  203.         if ( pWin32ServiceCall )
  204.             pWin32ServiceCall->fIgnore = TRUE;
  205.     }
  206.     
  207.     fclose( pFile );
  208.     
  209.     return TRUE;
  210. }
  211.